"""
类名：LoginApi
实例属性：
login_url：保存登陆的url，"http://ihrm-test.itheima.net/api/sys/login"
实例方法：
    def login(self, body)：
        # 功能：发送登陆请求
        # :param body: 发送请求的请求体
        # :return: 发送登陆请求后的响应对象
    def get_au_headers(self, body):
        # 功能：获取认证请求头
        # :param body: 登陆的请求体数据
        # :return: 成功登陆的认证请求头
"""
import requests
from config import URL


class LoginApi(object):
    def __init__(self):
        self.login_url = URL + "/api/sys/login"

    def login(self, body):
        resp = requests.post(self.login_url, json=body)
        return resp

    def get_au_headers(self, body):
        # 1.测试数据
        # 2.通过实例化对象self调用login方法，获取响应对象
        resp = self.login(body)
        # 3.获取响应内容的json格式
        json_data = resp.json()
        print(f'LoginApi 提取登陆成功的token返回的内容 = {json_data}')
        # 4.根据接口文档，组装一个认证的token请求头
        au_headers = {'Authorization': json_data.get('data')}
        # 5.返回组装的headers
        return au_headers
